package com.hcl.tree.binarytree;

/**
 * 二叉树
 */
public class BinaryTreeDemo {
    public static void main(String[] args) {

    }
}

class BinaryTree {
    // 根节点
    private HeroNode root;

    public void setRoot(HeroNode root) {
        this.root = root;
    }

    // 前序遍历
    public void preOrder(){
        if (root!=null){
            this.root.preOrder();
        } else {
            System.out.println("二叉树为空，不能遍历！");
        }
    }

    // 中序遍历
    public void infixOrder(){
        if (root!=null){
            this.root.infixOrder();
        } else {
            System.out.println("二叉树为空，不能遍历！");
        }
    }

    // 后序遍历
    public void postOrder(){
        if (root!=null){
            this.root.postOrder();
        } else {
            System.out.println("二叉树为空，不能遍历！");
        }
    }
}

class HeroNode {
    private String no;
    private String name;
    private HeroNode left; // 左子节点
    private HeroNode right; // 右子节点

    // 前序遍历 先输入当前节点；如果左子节点不为空，递归前序遍历；如果右子节点不为空，递归前序遍历
    public void preOrder(){
        // 先输出父节点
        System.out.println(this);
        // 递归左子树前序遍历
        if (this.left !=null){
            this.left.preOrder();
        }
        // 递归右子树前序遍历
        if (this.right !=null){
            this.right.preOrder();
        }
    }
    // 中序遍历
    public void infixOrder(){
        // 递归左子树中序遍历
        if (this.left!=null){
            this.left.infixOrder();
        }
        // 输出当前节点
        System.out.println(this);
        // 递归右子树中序遍历
        if (this.right!=null){
            this.right.infixOrder();
        }
    }

    // 后续遍历 如果左子节点不为空，递归左子节点；如果右子节点不为空，递归右子节点；输出当前节点
    public void postOrder(){
        if (this.left!=null){
            this.left.postOrder();
        }
        if (this.right!=null){
            this.right.postOrder();
        }
        System.out.println(this);
    }


    @Override
    public String toString() {
        return "HeroNode{" +
                "no='" + no + '\'' +
                ", name='" + name + '\'' +
                '}';
    }

    public HeroNode(String no, String name) {
        this.no = no;
        this.name = name;
    }

    public String getNo() {
        return no;
    }

    public void setNo(String no) {
        this.no = no;
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    public HeroNode getLeft() {
        return left;
    }

    public void setLeft(HeroNode left) {
        this.left = left;
    }

    public HeroNode getRight() {
        return right;
    }

    public void setRight(HeroNode right) {
        this.right = right;
    }
}
